
from django.template import Library

register = Library()


@register.simple_tag
def use_space(size):
    if size >= 1024 * 1024 * 1024:
        # 大于GB是，转成GB 保留两位小数
        return '%.2f GB' % (size / (1024 * 1024 * 1024), )
    elif size >= 1024 * 1024:
        return '%.2f MB' % (size / (1024 * 1024), )
    elif size >= 1024:
        return '%.2f KB' % (size / 1024, )
    else:
        return '%d B' % size
